package com.ruoyi.ade.llm.model;

public class AdeKnowledgeRerankConfig {

    //private String model;  //暂不开发 默认一个模型
    /**
     * 相关性阈值
     */
    private float reScoreThreshold;

    private String rerankModel;

    /**
     * top N
     */
    private int topN;

    public static AdeKnowledgeRerankConfig getDefault(){
        return new AdeKnowledgeRerankConfig(RerankRequest.model_rerank_bce_reranker_base,0.6f,10);
    }

    public AdeKnowledgeRerankConfig(){}
    public AdeKnowledgeRerankConfig(String rerankModel,float reScoreThreshold, int topN) {
        this.rerankModel = rerankModel;
        this.reScoreThreshold = reScoreThreshold;
        this.topN = topN;
    }

    public String getRerankModel() {
        return rerankModel;
    }

    public void setRerankModel(String rerankModel) {
        this.rerankModel = rerankModel;
    }

    public float getReScoreThreshold() {
        return reScoreThreshold;
    }

    public void setReScoreThreshold(float reScoreThreshold) {
        this.reScoreThreshold = reScoreThreshold;
    }

    public int getTopN() {
        return topN;
    }

    public void setTopN(int topN) {
        this.topN = topN;
    }
}
